# source /home/work_nfs8/xlgeng/new_workspace/.bashrc
# my_named_nohup train_zipformer ./zipformer/train.py --world-size 4
# export CUDA_VISIBLE_DEVICES="2"
# ./zipformer/decode.py \
#     --epoch 30 \
#     --avg 5 \
#     --exp-dir ./zipformer/exp \
#     --lang-dir data/lang_char \
#     --max-duration 600 \
#     --decoding-method modified_beam_search \
#     --beam-size 4

#export PYTHONPATH=/home/work_nfs8/xlgeng/new_workspace/icefall:$PYTHONPATH
export PYTHONPATH=./zipformer:$PYTHONPATH
export CUDA_VISIBLE_DEVICES="0,1"
./zipformer/train.py \
  --world-size 2 \
  --num-epochs 20 \
  --exp-dir zipformer/exp_common_new_stage \
  --train-fbank-dir /home/work_nfs8/xlgeng/new_workspace/gxl_ai_utils/eggs/cats_and_dogs/icefall_assistant/en_cn/gxl_data/13000h/fbank_common \
  --valid-fbank-dir /home/work_nfs8/xlgeng/new_workspace/gxl_ai_utils/eggs/cats_and_dogs/icefall_assistant/en_cn/gxl_data/3000h/fbank_common \
  --test-parent-dir /home/work_nfs8/xlgeng/new_workspace/gxl_ai_utils/eggs/cats_and_dogs/icefall_assistant/en_cn/gxl_data/test_family \
  --train_prefix 13000h \
  --valid_prefix 3000h \
  --lr-epochs 3.5 \
  --max-duration 420 \
  --base-lr 0.045 \
  --bpe-model /home/work_nfs8/xlgeng/new_workspace/icefall/egs/multi_zh_en/ASR/gxl_data/lang_bbpe_2000/bbpe.model \
  --use-transducer True \
  --use-ctc True \
  --checkpoint-path /home/work_nfs8/xlgeng/new_workspace/gxl_ai_utils/eggs/cats_and_dogs/icefall_assistant/en_cn/zipformer/exp_common_new_stage/checkpoint-4000.pt
  # --start-batch 4000
    # --start-epoch 1 \
  # --checkpoint-path /home/work_nfs8/xlgeng/new_workspace/checkpoint/zipformer/epoch-57.pt
    # --use-fp16 1 \
#    --causal 1 \
#  --on-the-fly-feats true
